// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_COMMON_CROSS_SITE_DOCUMENT_CLASSIFIER_H_
#define CONTENT_COMMON_CROSS_SITE_DOCUMENT_CLASSIFIER_H_

#include "base/macros.h"
#include "base/strings/string_piece.h"
#include "content/common/content_export.h"
#include "url/gurl.h"
#include "url/origin.h"

namespace content {

// CrossSiteDocumentClassifier implements the cross-site document blocking
// policy (XSDP) for Site Isolation. XSDP will monitor network responses to a
// renderer and block illegal responses so that a compromised renderer cannot
// steal private information from other sites.

enum CrossSiteDocumentMimeType {
    // Note that these values are used in histograms, and must not change.
    CROSS_SITE_DOCUMENT_MIME_TYPE_HTML = 0,
    CROSS_SITE_DOCUMENT_MIME_TYPE_XML = 1,
    CROSS_SITE_DOCUMENT_MIME_TYPE_JSON = 2,
    CROSS_SITE_DOCUMENT_MIME_TYPE_PLAIN = 3,
    CROSS_SITE_DOCUMENT_MIME_TYPE_OTHERS = 4,
    CROSS_SITE_DOCUMENT_MIME_TYPE_MAX,
};

class CONTENT_EXPORT CrossSiteDocumentClassifier {
public:
    // Returns the representative mime type enum value of the mime type of
    // response. For example, this returns the same value for all text/xml mime
    // type families such as application/xml, application/rss+xml.
    static CrossSiteDocumentMimeType GetCanonicalMimeType(
        const std::string& mime_type);

    // Returns whether this scheme is a target of cross-site document
    // policy(XSDP). This returns true only for http://* and https://* urls.
    static bool IsBlockableScheme(const GURL& frame_origin);

    // Returns whether the two urls belong to the same sites.
    static bool IsSameSite(const url::Origin& frame_origin,
        const GURL& response_url);

    // Returns whether there's a valid CORS header for frame_origin.  This is
    // simliar to CrossOriginAccessControl::passesAccessControlCheck(), but we use
    // sites as our security domain, not origins.
    // TODO(dsjang): this must be improved to be more accurate to the actual CORS
    // specification. For now, this works conservatively, allowing XSDs that are
    // not allowed by actual CORS rules by ignoring 1) credentials and 2)
    // methods. Preflight requests don't matter here since they are not used to
    // decide whether to block a document or not on the client side.
    static bool IsValidCorsHeaderSet(const url::Origin& frame_origin,
        const GURL& website_origin,
        const std::string& access_control_origin);

    static bool SniffForHTML(base::StringPiece data);
    static bool SniffForXML(base::StringPiece data);
    static bool SniffForJSON(base::StringPiece data);

private:
    CrossSiteDocumentClassifier(); // Not instantiable.

    DISALLOW_COPY_AND_ASSIGN(CrossSiteDocumentClassifier);
};

} // namespace content

#endif // CONTENT_COMMON_CROSS_SITE_DOCUMENT_CLASSIFIER_H_
